numpy
matplotlib
tqdm
torch